package com.mall.framework.web.domain;

import com.mall.common.utils.Arith;
import com.mall.common.utils.IpUtils;
import com.mall.framework.web.domain.server.*;
import java.net.UnknownHostException;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;
import oshi.SystemInfo;
import oshi.hardware.CentralProcessor;
import oshi.hardware.CentralProcessor.TickType;
import oshi.hardware.GlobalMemory;
import oshi.hardware.HardwareAbstractionLayer;
import oshi.software.os.FileSystem;
import oshi.software.os.OSFileStore;
import oshi.software.os.OperatingSystem;
import oshi.util.Util;

/**
 * 服务器相关信息
 * Created by zhonglin on 2020/4/8.
 */
public class Server {

  private static final int OSHI_WAIT_SECOND = 1000;

  /**
   * CPU相关信息
   */
  private Cpu cpu = new Cpu();

  /**
   * 內存相关信息
   */
  private Mem mem = new Mem();

  /**
   * JVM相关信息
   */
  private Jvm jvm = new Jvm();

  /**
   * 服务器相关信息
   */
  private Sys sys = new Sys();

  /**
   * 磁盘相关信息
   */
  private List<SysFile> sysFiles = new LinkedList<SysFile>();

  public Cpu getCpu() {

    return cpu;
  }

  public void setCpu(Cpu cpu) {

    this.cpu = cpu;
  }

  public Mem getMem() {

    return mem;
  }

  public void setMem(Mem mem) {

    this.mem = mem;
  }

  public Jvm getJvm() {

    return jvm;
  }

  public void setJvm(Jvm jvm) {

    this.jvm = jvm;
  }

  public Sys getSys() {

    return sys;
  }

  public void setSys(Sys sys) {

    this.sys = sys;
  }

  public List<SysFile> getSysFiles() {

    return sysFiles;
  }

  public void setSysFiles(List<SysFile> sysFiles) {

    this.sysFiles = sysFiles;
  }

  public void copyTo()
      throws Exception {

    SystemInfo               si  = new SystemInfo();
    HardwareAbstractionLayer hal = si.getHardware();

    setCpuInfo(hal.getProcessor());

    setMemInfo(hal.getMemory());

    setSysInfo();

    setJvmInfo();

    setSysFiles(si.getOperatingSystem());
  }

  /**
   * 设置CPU信息
   */
  private void setCpuInfo(CentralProcessor processor) {
    // CPU信息
    long[] prevTicks = processor.getSystemCpuLoadTicks();
    Util.sleep(OSHI_WAIT_SECOND);
    long[] ticks    = processor.getSystemCpuLoadTicks();
    long   nice     = ticks[TickType.NICE.getIndex()] - prevTicks[TickType.NICE.getIndex()];
    long   irq      = ticks[TickType.IRQ.getIndex()] - prevTicks[TickType.IRQ.getIndex()];
    long   softirq  = ticks[TickType.SOFTIRQ.getIndex()] - prevTicks[TickType.SOFTIRQ.getIndex()];
    long   steal    = ticks[TickType.STEAL.getIndex()] - prevTicks[TickType.STEAL.getIndex()];
    long   cSys     = ticks[TickType.SYSTEM.getIndex()] - prevTicks[TickType.SYSTEM.getIndex()];
    long   user     = ticks[TickType.USER.getIndex()] - prevTicks[TickType.USER.getIndex()];
    long   iowait   = ticks[TickType.IOWAIT.getIndex()] - prevTicks[TickType.IOWAIT.getIndex()];
    long   idle     = ticks[TickType.IDLE.getIndex()] - prevTicks[TickType.IDLE.getIndex()];
    long   totalCpu = user + nice + cSys + idle + iowait + irq + softirq + steal;
    cpu.setCpuNum(processor.getLogicalProcessorCount());
    cpu.setTotal(totalCpu);
    cpu.setSys(cSys);
    cpu.setUsed(user);
    cpu.setWait(iowait);
    cpu.setFree(idle);
  }

  /**
   * 设置内存信息
   */
  private void setMemInfo(GlobalMemory memory) {

    mem.setTotal(memory.getTotal());
    mem.setUsed(memory.getTotal() - memory.getAvailable());
    mem.setFree(memory.getAvailable());
  }

  /**
   * 设置服务器信息
   */
  private void setSysInfo() {

    Properties props = System.getProperties();
    sys.setComputerName(IpUtils.getHostName());
    sys.setComputerIp(IpUtils.getHostIp());
    sys.setOsName(props.getProperty("os.name"));
    sys.setOsArch(props.getProperty("os.arch"));
    sys.setUserDir(props.getProperty("user.dir"));
  }

  /**
   * 设置Java虚拟机
   */
  private void setJvmInfo()
      throws UnknownHostException {

    Properties props = System.getProperties();
    jvm.setTotal(Runtime.getRuntime()
                        .totalMemory());
    jvm.setMax(Runtime.getRuntime()
                      .maxMemory());
    jvm.setFree(Runtime.getRuntime()
                       .freeMemory());
    jvm.setVersion(props.getProperty("java.version"));
    jvm.setHome(props.getProperty("java.home"));
  }

  /**
   * 设置磁盘信息
   */
  private void setSysFiles(OperatingSystem os) {

    FileSystem    fileSystem = os.getFileSystem();
    OSFileStore[] fsArray    = fileSystem.getFileStores();
    for (OSFileStore fs : fsArray) {
      long    free    = fs.getUsableSpace();
      long    total   = fs.getTotalSpace();
      long    used    = total - free;
      SysFile sysFile = new SysFile();
      sysFile.setDirName(fs.getMount());
      sysFile.setSysTypeName(fs.getType());
      sysFile.setTypeName(fs.getName());
      sysFile.setTotal(convertFileSize(total));
      sysFile.setFree(convertFileSize(free));
      sysFile.setUsed(convertFileSize(used));
      sysFile.setUsage(Arith.mul(Arith.div(used, total, 4), 100));
      sysFiles.add(sysFile);
    }
  }

  /**
   * 字节转换
   *
   * @param size 字节大小
   *
   * @return 转换后值
   */
  public String convertFileSize(long size) {

    long kb = 1024;
    long mb = kb * 1024;
    long gb = mb * 1024;
    if (size >= gb) {
      return String.format("%.1f GB", (float) size / gb);
    } else if (size >= mb) {
      float f = (float) size / mb;
      return String.format(f > 100
                               ? "%.0f MB"
                               : "%.1f MB", f);
    } else if (size >= kb) {
      float f = (float) size / kb;
      return String.format(f > 100
                               ? "%.0f KB"
                               : "%.1f KB", f);
    } else {
      return String.format("%d B", size);
    }
  }

}
